# export_onnx.py
from ultralytics import YOLO

model = YOLO(r"D:\NCNN\ncnn-20230816-windows-vs2017\x64\bin\best31n.pt")
model.export(
    format="onnx",
    batch=1,              # 固定batch=1
    imgsz=(320, 320),     # 明确输入尺寸
    opset=12,             # 必须≤13（ncnn兼容性）
    simplify=True,        # 合并冗余节点
    dynamic=False,        # 固定所有维度
    device="cpu",         # 避免CUDA依赖
    nms=True,           # 禁用NMS（关键！）
)
print("✅ ONNX导出完成，请用Netron验证输出层是否为[1,2100,7]")




